package com.example.binarytree;

import com.example.structure.TreeNode;

/**
 * 450. 删除二叉搜索树中的节点----难难难！
 * 给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。
 * <p>
 * 一般来说，删除节点可分为两个步骤：
 * <p>
 * 首先找到需要删除的节点；
 * 如果找到了，删除它。
 * <p>
 * <p>
 * 示例 1:
 * <p>
 * 输入：root = [5,3,6,2,4,null,7], key = 3
 * 输出：[5,4,6,2,null,null,7]
 * 解释：给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。
 * 一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
 * 另一个正确答案是 [5,2,6,null,4,null,7]。
 */
public class DeleteNode {
    public TreeNode deleteNode(TreeNode root, int key) {
        if (root == null) return null;
        if (key > root.val)
            root.right = deleteNode(root.right, key); // 去右子树删除
        else if (key < root.val)
            root.left = deleteNode(root.left, key);  // 去左子树删除
        else {  // 当前节点就是要删除的节点
            if (root.left == null) return root.right;      // 情况1，欲删除节点无左子
            else if (root.right == null) return root.left;  // 情况2，欲删除节点无右子
            else if (root.left != null && root.right != null) {  // 情况3，欲删除节点左右子都有
                TreeNode node = root.right;
                while (node.left != null) {      // 寻找欲删除节点右子树的最左节点
                    node = node.left;
                }
                node.left = root.left;     // 将欲删除节点的左子树成为其右子树的最左节点的左子树
                root = root.right;         // 欲删除节点的右子顶替其位置，节点被删除
            }
        }
        return root;
    }
}
